package net.xiaosaguo.mymall.search.config;

import org.apache.http.HttpHost;
import org.elasticsearch.client.RequestOptions;
import org.elasticsearch.client.RestClient;
import org.elasticsearch.client.RestClientBuilder;
import org.elasticsearch.client.RestHighLevelClient;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

/**
 * description: Elasticsearch 配置
 * <p>
 * 1. 导入依赖
 * 2. 编写配置，在 spring 容器中注入 esRestClient 实例
 * 3. 操纵 es
 *
 * @author xiaosaguo
 * @date 2020/06/19
 */
@Configuration
public class ElasticsearchConfig {

    public static final RequestOptions COMMON_OPTIONS;
    static {
        RequestOptions.Builder builder = RequestOptions.DEFAULT.toBuilder();
        // builder.addHeader("Authorization", "Bearer " + TOKEN);
        // builder.setHttpAsyncResponseConsumerFactory(
        //         new HttpAsyncResponseConsumerFactory
        //                 .HeapBufferedResponseConsumerFactory(30 * 1024 * 1024 * 1024));
        COMMON_OPTIONS = builder.build();
    }

    @Bean
    public RestHighLevelClient esRestClient() {

        // 自己试写
        RestClientBuilder builder = null;
        builder = RestClient.builder(new HttpHost("192.168.235.131", 9200, "http"));
        RestHighLevelClient client = new RestHighLevelClient(builder);

        // 官方文档
        // RestHighLevelClient client = new RestHighLevelClient(
        //         RestClient.builder(
        //                 new HttpHost("192.168.235.131", 9200, "http")
        //         )
        // );

        return client;
    }
}
